package backtrack;

/**
 * @author kaho
 * @since 2021/1/4
 */
public class _52_N皇后II {
    static int[] queue;
    static int way;


    /**
     * 给定一个整数 n，返回 n 皇后不同的解决方案的数量。
     * @param n
     * @return
     */
    public static int totalNQueens(int n) {
        if(n<1){
            return 0;
        }
        queue = new int[n];
        place(0);
        return way;
    }

    private static void place(int row) {
        if (row == queue.length) {
            way++;
            return;
        }
        for (int col = 0; col < queue.length; col++) {
            if(isValid(row, col)){
                queue[row] = col;
                place(row + 1);
            }
        }
    }

    private static boolean isValid(int row, int col) {
        for (int i = 0; i < row; i++) {
            if (queue[i] == col) {
                return false;
            }
            if (row-i == Math.abs(col-queue[i])) {
                return false;
            }
        }
        return true;
    }

    public static void main(String[] args) {
        int i = totalNQueens(1);
    }
}
